package com.tgr.shortLink.project.toolkit;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;

import java.io.IOException;

/**
 * 描述：
 * 作者：小陶不慌张
 * 文件：FaviconUtil
 * 日期：2024/1/20 22:22
 */

public class FaviconUtil {
    public static String getFaviconUrl(String websiteUrl) throws IOException {
        Document doc = Jsoup.connect(websiteUrl).get();
        Element faviconElement = doc.selectFirst("link[href~=.*\\.(ico|png)]");

        if (faviconElement != null) {
            String faviconUrl = faviconElement.attr("href");
            if (!faviconUrl.startsWith("http")) {
                faviconUrl = websiteUrl + faviconUrl;
            }
            return faviconUrl;
        }

        return null;
    }
}
